#ifndef VOLUMEWIDGET_H
#define VOLUMEWIDGET_H

#include "AmaranthusCommon.h"
#include "VisWidgetBase.h"

#include "vtkSmartPointer.h"
#include "vtkActor.h"
#include "vtkProp3D.h"
#include "vtkPolyDataMapper.h"
#include "vtkImageDataGeometryFilter.h"
#include "vtkLineSource.h"

#include <QVTKWidget.h>
#include <QMap>

class VolumeWidget : public VisWidgetBase
{
    Q_OBJECT

public:
    typedef vtkSmartPointer<vtkLineSource> LinePointer;

    VolumeWidget(QWidget* parent = NULL)
    { m_WidgetType = VOLUME; }

    ~VolumeWidget() {}

    void SetVolume(Material *material, PolyDataPointer polyData);
    void DisplayBox(double x, double y, double z);
};

#endif
